from customize.get_ollama import GetOllama
from debate_agents.rag_debate_agent import RAGDebateAgent
from my_tools.debate_evidence_retriever import DebateEvidenceTool
llm = GetOllama(ip="10.12.8.21:11434", model_name="deepseek-r1:14b", model_type=1)()
llm_mini = GetOllama(model_name="qwen2.5:3b", model_type=1)()
debate_agent = RAGDebateAgent(llm=llm, evidence_tool=DebateEvidenceTool(llm=llm_mini))
prompt = "在开篇立论中，需清晰、有条理地率先阐述己方核心观点与主要论据，构建起坚实完整的逻辑框架。可适当引用权威数据、名言警句等增强可信度，如 “据权威机构报告，超 70% 的企业因创新实现利润增长，证明创新是企业发展核心动力”。注意语言简洁，观点明确，引用准确，避免冗长模糊。作为先发言方，论点要坚实，逻辑严密，不给反方留明显漏洞，论据要有广泛代表性与权威性，为后续辩论筑牢根基。"
print(debate_agent.invoke(side="正方", topic="手机应该进入中学校园", stage="开篇立论阶段", role="一辩", task="开篇立论",
                          role_prompt=prompt))